# 获取到模型
import os

from langchain_community.embeddings import DashScopeEmbeddings
from langchain_openai import ChatOpenAI

# 模型实例
llm = ChatOpenAI(
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    model="qwen-plus",
    temperature=0.8
)

# 向量模型实例
embeddings = DashScopeEmbeddings(
    model="text-embedding-v4",  # 固定模型名称，不可修改
    dashscope_api_key=os.getenv("DASHSCOPE_API_KEY")  # 传入你的API密钥
)


def get_embedding(text):
    """
    获取传递来的文本的向量
    :param text:
    :return:
    """
    return embeddings.embed_query(text)
